from pydantic import BaseModel, Field
from typing import Optional
from src.utils.response import ResponseModel

class DraftRequest(BaseModel):
    """创建草稿请求模型"""
    width: int = Field(default=1080, description="视频宽度")
    height: int = Field(default=1920, description="视频高度")

class DraftData(BaseModel):
    """草稿数据模型"""
    draft_id: str = Field(description="草稿ID")
    draft_url: str = Field(description="草稿URL")

class DraftQueryRequest(BaseModel):
    """草稿数据模型"""
    draft_id: str = Field(description="草稿ID")
    force_update: Optional[bool] = Field(default = True,description="强制更新")

class DraftSaveRequest(BaseModel):
    """草稿数据模型"""
    draft_id: str = Field(description="草稿ID")
    draft_dir: str = Field(description="剪映草稿目录")

class DraftDataResponse(ResponseModel[DraftData]):
    """草稿数据响应模型"""
    pass